package bintree.leetcode;

import java.util.LinkedList;
import java.util.Queue;

public class IsCompleteTree {
    public boolean isCompleteTree(TreeNode root) {
        if (root == null) {
            return true;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        // 引入标志位区分第一还是第二阶段
        boolean isSecondStep = false;
        while (!queue.isEmpty()) {
            TreeNode cur = queue.poll();
            // 此时访问节点是看当前节点树是否满足完全二叉树概念
            if (!isSecondStep) {
                if (cur.left != null && cur.right != null) {
                    // 直接入队列进行下次循环
                    queue.offer(cur.left);
                    queue.offer(cur.right);
                }else if (cur.left == null && cur.right != null) {
                    // 只有右树，没有左树，反例
                    return false;
                }else if (cur.left != null) {
                   // 此时节点只有左树，将左树入队，切换状态
                   queue.offer(cur.left);
                   isSecondStep = true;
                }else {
                    // 该节点是第一个叶子节点，切换状态
                    isSecondStep = true;
                }
            }else {
                // 第二阶段，所有节点都是叶子节点，不能有子树
                if (cur.left != null || cur.right != null) {
                    // 反例
                    return false;
                }
            }
        }
        // 当层序遍历结束仍然没找到反例，说明整棵树是一个完全二叉树
        return true;
    }
}
